import torch
import numpy as np
import cv2

data = cv2.imread("test.png")

print(data)

cv2.imshow("test", data)
cv2.waitKey(0)
# cv2.imshow("test", im_data)
# cv2.waitKey(0)
# ba = np.zeros([2, 2])

out = torch.from_numpy(data)
print(out)

